# 生成自己图像的train_list.txt文件或者test_list.txt文件
# 日期: 2020/07/20

import os
import sys
import glob
path = 'F:/laibo/data_hua_711_tu/test_img_and_txt'

img_name = [img_path for img_path in glob.glob(path+'/*.jpg')]
# print(img_name)

if __name__ == '__main__':
    with open(path+'/test_list.txt', 'w') as f:
        for i in range(len(img_name)):
            f.write(img_name[i])
            f.write('\n')
